Upload app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from scholarly import scholarly
+from scholarly import scholarly, ProxyGenerator
 import openai
 
 def process(key, choice, artitle, trans):
@@ -7,6 +7,9 @@ def process(key, choice, artitle, trans):
     
     results = []
     if choice=='单个生成':
+        pg = ProxyGenerator()
+        pg.FreeProxies()
+        scholarly.use_proxy(pg)
         search_query = scholarly.search_pubs(str(artitle))
         pub = next(search_query)
         bib = scholarly.bibtex(pub)
@@ -21,17 +24,22 @@ def process(key, choice, artitle, trans):
                 {"role": "system", "content": "You are a helpful assistant."},
                 {"role": "user", "content": prompt}
             ]
-            ) 
-            print(completion.choices[0].message["content"])
-            print('-------------')
+            )
+            print(completion.choices[0].message['content'])
             print(completion.choices[0].text)
-            results.append(completion.choices[0].message["content"])
+            results.append(completion.choices[0].message)
     if choice=='批量生成':
         m_artitle = artitle.split('\n')
         for i in range(len(m_artitle)):
             if trans=='bib':
                 results.append(bib)
             else:
+                pg = ProxyGenerator()
+                pg.FreeProxies()
+                scholarly.use_proxy(pg)
+                search_query = scholarly.search_pubs(str(m_artitle[i]))
+                pub = next(search_query)
+                bib = scholarly.bibtex(pub)
                 prompt = "请把以下bib格式转为"+str(trans)+"格式："+str(bib)
                 completion = openai.ChatCompletion.create(
                 model="gpt-3.5-turbo",
@@ -73,4 +81,4 @@ demo = gr.Interface(fn=process,
                     outputs="text",
                     title=title,
                     description=description)
-demo.launch()
+demo.launch(share=True)