Update app.py

app.py

@@ -85,8 +85,30 @@ For example, if asked "What is the capital of France?", respond simply with "Par
                     print(f"Raw response type: {type(raw_response)}")
                     print(f"Raw response: {raw_response}")
                 
-                # Handle ALL possible response formats with recursive flattening
-                answer = self._extract_answer(raw_response)
+                # Handle ALL possible response formats
+                if isinstance(raw_response, dict):
+                    answer = raw_response.get('choices', [{}])[0].get('message', {}).get('content', str(raw_response))
+                elif isinstance(raw_response, list):
+                    if len(raw_response) > 0:
+                        if isinstance(raw_response[0], dict):
+                            # Common format: [{"role": "assistant", "content": "..."}]
+                            answer = raw_response[0].get('content', str(raw_response[0]))
+                        elif isinstance(raw_response[0], list):
+                            # Nested list - dig deeper
+                            nested = raw_response[0]
+                            if isinstance(nested, list) and len(nested) > 0:
+                                if isinstance(nested[0], dict):
+                                    answer = nested[0].get('content', str(nested[0]))
+                                else:
+                                    answer = str(nested[0])
+                            else:
+                                answer = str(raw_response[0])
+                        else:
+                            answer = str(raw_response[0])
+                    else:
+                        answer = "No response"
+                else:
+                    answer = str(raw_response)
                 
                 if self.verbose:
                     print(f"Extracted answer type: {type(answer)}")
@@ -112,71 +134,28 @@ For example, if asked "What is the capital of France?", respond simply with "Par
                 traceback.print_exc()  
             return error_msg
     
-    def _extract_answer(self, response):
+    def _clean_answer(self, answer: any) -> str:
         """
-        Recursively extract the answer from any nested structure.
-        This handles dict, list, and nested combinations.
+        Ultra-safe answer extraction and cleaning
         """
-        if response is None:
-            return ""
-        
-        # If it's already a string, return it
-        if isinstance(response, str):
-            return response
-        
-        # If it's a dict, try to get 'content' or 'message'
-        if isinstance(response, dict):
-            # Try common keys
-            for key in ['content', 'message', 'text', 'answer']:
-                if key in response:
-                    return self._extract_answer(response[key])
-            # Try OpenAI-style format
-            if 'choices' in response:
-                choices = response['choices']
-                if choices and len(choices) > 0:
-                    return self._extract_answer(choices[0])
-            # If nothing found, convert to string
-            return str(response)
-        
-        # If it's a list, recursively process elements
-        if isinstance(response, list):
-            if len(response) == 0:
-                return ""
-            # Try to extract from first element
-            first_element = self._extract_answer(response[0])
-            # If first element is empty or None, try other elements
-            if not first_element and len(response) > 1:
-                for element in response[1:]:
-                    extracted = self._extract_answer(element)
-                    if extracted:
-                        return extracted
-            return first_element
-        
-        # For any other type, convert to string
-        return str(response)
-    
-    def _clean_answer(self, answer) -> str:
-        """
-        Ultra-safe answer extraction and cleaning.
-        Ensures the answer is always a string.
-        """
-        # First, ensure we have a string
+        # Force to string immediately with extra safety
         try:
-            # Recursively extract if needed
-            if not isinstance(answer, str):
-                answer = self._extract_answer(answer)
-            
-            # At this point, answer should be a string
+            if answer is None:
+                return ""
+            if isinstance(answer, list):
+                # If it's a list, try to extract meaningful content
+                if len(answer) == 0:
+                    return ""
+                # Try to get content from first element
+                answer = answer[0] if len(answer) > 0 else ""
+            if isinstance(answer, dict):
+                # If it's a dict, try to get 'content' or convert to string
+                answer = answer.get('content', str(answer))
             if not isinstance(answer, str):
                 answer = str(answer)
-            
         except Exception as e:
             print(f"Error in initial conversion: {e}")
-            traceback.print_exc()
-            try:
-                return str(answer) if answer else ""
-            except:
-                return ""
+            return str(answer) if answer else ""
         
         # Now answer should definitely be a string
         try:
@@ -201,21 +180,23 @@ For example, if asked "What is the capital of France?", respond simply with "Par
             
             return answer
         except Exception as e:
-            print(f"Error in answer cleaning: {e}")
-            traceback.print_exc()
-            try:
-                return str(answer) if answer else ""
-            except:
-                return ""
+            print(f"Error in answer cleaning: {e}, returning raw string")
+            return str(answer) if answer else ""
 
 def run_and_submit_all(profile: gr.OAuthProfile | None):
     """
     Fetches all questions, runs the BasicAgent on them, submits all answers,
     and displays the results.
     """
-    if profile is None:
-        return "Please log in with your Hugging Face account first.", None
-    username = profile.username
+    space_id = os.getenv("SPACE_ID")
+
+    if profile:
+        username = f"{profile.username}"
+        print(f"User logged in: {username}")
+    else:
+        print("User not logged in.")
+        return "Please Login to Hugging Face with the button.", None
+
     api_url = DEFAULT_API_URL
     questions_url = f"{api_url}/questions"
     submit_url = f"{api_url}/submit"
@@ -274,7 +255,6 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
         except Exception as e:
             print(f"Error running agent on task {task_id}: {e}")
-            traceback.print_exc()
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": f"AGENT ERROR: {e}"})
 
     if not answers_payload:
@@ -326,7 +306,6 @@ def run_and_submit_all(profile: gr.OAuthProfile | None):
     except Exception as e:
         status_message = f"An unexpected error occurred during submission: {e}"
         print(status_message)
-        traceback.print_exc()
         results_df = pd.DataFrame(results_log)
         return status_message, results_df
 
@@ -382,7 +361,7 @@ class Tee:
 
 if __name__ == "__main__":
     # Redirect stdout and stderr
-    log_timestamp = datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
+    log_timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
     log_file = f"./logs/output_{log_timestamp}.log"
     tee = Tee(log_file)
     sys.stdout = tee