Update app.py

app.py

@@ -476,7 +476,7 @@ def get_response_from_llama(query, model, selected_docs, file_type, num_calls=1,
     
     client = InferenceClient(model, token=huggingface_token)
     logging.info("InferenceClient initialized")
-    
+ 
     if file_type == "excel":
         # Excel functionality
         system_instruction = """You are a highly specialized Python programmer with deep expertise in data analysis and visualization using Excel spreadsheets. 
@@ -496,7 +496,7 @@ def get_response_from_llama(query, model, selected_docs, file_type, num_calls=1,
             {"role": "system", "content": system_instruction},
             {"role": "user", "content": f"Based on the following data extracted from Excel spreadsheets:\n{context}\n\nPlease provide the Python code needed to execute the following task: '{query}'. Ensure that the code is derived directly from the dataset. If a chart is requested, use the matplotlib library to generate the appropriate visualization."}
         ]
-        
+    
     elif file_type == "pdf":
         # PDF functionality
         embed = get_embeddings()
@@ -519,15 +519,15 @@ def get_response_from_llama(query, model, selected_docs, file_type, num_calls=1,
     else:
         raise ValueError("Invalid file type. Use 'excel' or 'pdf'.")
       
-    logging.info(f"Prepared messages: {messages}")
-    
+#    logging.info(f"Prepared messages: {messages}")
+ 
     full_response = ""
     for i in range(num_calls):
         logging.info(f"Starting API call {i+1}/{num_calls}")
         try:
-            for message in client.chat.completions.create(
+            for message in client.chat.completion(
                 messages=messages,
-                max_tokens=2000,
+                max_tokens=2048,
                 temperature=temperature,
                 stream=True,   
             ):
@@ -858,7 +858,7 @@ demo = gr.ChatInterface(
     ],
     title="AI-powered PDF Chat and Web Search Assistant",
     description="Chat with your PDFs or use web search to answer questions.",
-    theme=gr.Theme.from_hub("JohnSmith9982/small_and_pretty"),
+    theme=gr.Theme.from_hub("allenai/gradio-theme"),
     css=css,
     examples=[
         ["Tell me about the contents of the uploaded PDFs."],