matteopilotto commited on
Commit
c1fdf4f
·
verified ·
1 Parent(s): ab66226

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -7
app.py CHANGED
@@ -2,16 +2,14 @@ import os
2
  import requests
3
  from datetime import datetime
4
 
5
-
6
  import gradio as gr
7
  from gradio import components
8
 
9
- from langchain.chat_models import ChatOpenAI
 
10
  from langchain.chains.summarize import load_summarize_chain
11
  from langchain.docstore.document import Document
12
 
13
- from openai.error import AuthenticationError
14
-
15
  def comment_datetime(comment_json):
16
  dt = datetime.fromisoformat(comment_json["created_at"][:-1])
17
  return comment_json["user"]["login"] + " commented on " + dt.strftime("%b %d, %Y") + ":"
@@ -79,7 +77,7 @@ def get_issue_info(url):
79
 
80
  def init_chain(openai_api_key):
81
  llm = ChatOpenAI(
82
- model_name="gpt-3.5-turbo-16k",
83
  temperature=0.0,
84
  openai_api_key=openai_api_key
85
  )
@@ -91,7 +89,7 @@ def init_chain(openai_api_key):
91
 
92
  def get_summary(text, chain):
93
  doc = Document(page_content=text)
94
- summary = chain.run([doc]).strip()
95
 
96
  return summary
97
 
@@ -113,8 +111,9 @@ def run_all(openai_api_key, url):
113
  try:
114
  chain = init_chain(openai_api_key)
115
  summary = get_summary(context, chain)
 
116
  return template_output.format(summary=summary, context=context)
117
- except (ValueError, AuthenticationError) as error:
118
  return "## Invalid OpenAI API key provided."
119
 
120
  html = """
 
2
  import requests
3
  from datetime import datetime
4
 
 
5
  import gradio as gr
6
  from gradio import components
7
 
8
+ import openai
9
+ from langchain_openai import ChatOpenAI
10
  from langchain.chains.summarize import load_summarize_chain
11
  from langchain.docstore.document import Document
12
 
 
 
13
  def comment_datetime(comment_json):
14
  dt = datetime.fromisoformat(comment_json["created_at"][:-1])
15
  return comment_json["user"]["login"] + " commented on " + dt.strftime("%b %d, %Y") + ":"
 
77
 
78
  def init_chain(openai_api_key):
79
  llm = ChatOpenAI(
80
+ model_name="gpt-4o",
81
  temperature=0.0,
82
  openai_api_key=openai_api_key
83
  )
 
89
 
90
  def get_summary(text, chain):
91
  doc = Document(page_content=text)
92
+ summary = chain.invoke([doc])["output_text"].strip()
93
 
94
  return summary
95
 
 
111
  try:
112
  chain = init_chain(openai_api_key)
113
  summary = get_summary(context, chain)
114
+ print(summary)
115
  return template_output.format(summary=summary, context=context)
116
+ except (ValueError, openai.AuthenticationError) as error:
117
  return "## Invalid OpenAI API key provided."
118
 
119
  html = """