Update app.py

app.py

@@ -205,6 +205,33 @@ class GradioInference:
             results["text"], max_length=150, min_length=30, do_sample=False
         )
 
+        ########################## PRUEBA CHATGPT #################################
+        from langchain.chains.llm import LLMChain
+        from langchain.prompts import PromptTemplate
+        from langchain.chains.combine_documents.stuff import StuffDocumentsChain
+        
+        OPENAI_API_KEY = os.environ['OPENAI_API_KEY']
+        
+        # Define prompt
+        prompt_template = """Write a concise summary of the following:
+        "{results["text"]}"
+        CONCISE SUMMARY:"""
+        prompt = PromptTemplate.from_template(prompt_template)
+        
+        # Define LLM chain
+        llm = ChatOpenAI(openai_api_key=OPENAI_API_KEY, temperature=0, model_name="gpt-3.5-turbo-16k")
+        llm_chain = LLMChain(llm=llm, prompt=prompt)
+        
+        # Define StuffDocumentsChain
+        stuff_chain = StuffDocumentsChain(
+            llm_chain=llm_chain, document_variable_name="text"
+        )
+        
+        docs = loader.load()
+        sum = stuff_chain.run(docs)
+
+        ########################## FIN PRUEBA CHATGPT #################################
+        
         #### Prueba
         WHITESPACE_HANDLER = lambda k: re.sub('\s+', ' ', re.sub('\n+', ' ', k.strip()))
         
@@ -270,7 +297,8 @@ class GradioInference:
         if lang == "english":
             return (
                 results["text"],
-                transcription_summary[0]["summary_text"],
+                sum,
+                # transcription_summary[0]["summary_text"],
                 formatted_keywords,
                 formatted_sentiment,
                 wordcloud_image,
@@ -278,7 +306,8 @@ class GradioInference:
         else:
             return (
                 results["text"],
-                summary,
+                sum,
+                # summary,
                 formatted_keywords,
                 formatted_sentiment,
                 wordcloud_image,