JeCabrera commited on
Commit
fbe527a
·
verified ·
1 Parent(s): 6f19708

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -10
app.py CHANGED
@@ -18,7 +18,7 @@ load_dotenv()
18
 
19
  # Configure Google Gemini API
20
  genai.configure(api_key=os.getenv('GOOGLE_API_KEY'))
21
- model = genai.GenerativeModel('gemini-1.5-flash')
22
 
23
  # Initialize session state variables if they don't exist
24
  if 'submitted' not in st.session_state:
@@ -65,8 +65,6 @@ with col1:
65
  uploaded_file = st.file_uploader("📄 Sube un archivo o imagen",
66
  type=['txt', 'pdf', 'docx', 'jpg', 'jpeg', 'png'])
67
 
68
- # Remove all the success message display code
69
-
70
  if uploaded_file is not None:
71
  file_type = uploaded_file.name.split('.')[-1].lower()
72
 
@@ -74,7 +72,6 @@ with col1:
74
  if file_type in ['txt', 'pdf', 'docx']:
75
  if file_type == 'txt':
76
  file_content = uploaded_file.read().decode('utf-8')
77
- # Remove success message code
78
 
79
  elif file_type == 'pdf':
80
  try:
@@ -83,7 +80,6 @@ with col1:
83
  file_content = ""
84
  for page in pdf_reader.pages:
85
  file_content += page.extract_text() + "\n"
86
- # Remove success message code
87
  except Exception as e:
88
  st.error(f"Error al leer el archivo PDF: {str(e)}")
89
  file_content = ""
@@ -93,14 +89,16 @@ with col1:
93
  import docx
94
  doc = docx.Document(uploaded_file)
95
  file_content = "\n".join([para.text for para in doc.paragraphs])
96
- # Remove success message code
97
  except Exception as e:
98
  st.error(f"Error al leer el archivo DOCX: {str(e)}")
99
  file_content = ""
100
 
101
- # Show preview with a header
102
- st.write("**Vista previa del contenido:**")
103
- st.write(file_content[:500] + "..." if len(file_content) > 500 else file_content)
 
 
 
104
 
105
  # Set file type flag
106
  is_image = False
@@ -117,7 +115,6 @@ with col1:
117
  "data": image_bytes
118
  }
119
  ]
120
- # Remove success message code
121
 
122
  # Set file type flag
123
  is_image = True
 
18
 
19
  # Configure Google Gemini API
20
  genai.configure(api_key=os.getenv('GOOGLE_API_KEY'))
21
+ model = genai.GenerativeModel('gemini-2.0-flash')
22
 
23
  # Initialize session state variables if they don't exist
24
  if 'submitted' not in st.session_state:
 
65
  uploaded_file = st.file_uploader("📄 Sube un archivo o imagen",
66
  type=['txt', 'pdf', 'docx', 'jpg', 'jpeg', 'png'])
67
 
 
 
68
  if uploaded_file is not None:
69
  file_type = uploaded_file.name.split('.')[-1].lower()
70
 
 
72
  if file_type in ['txt', 'pdf', 'docx']:
73
  if file_type == 'txt':
74
  file_content = uploaded_file.read().decode('utf-8')
 
75
 
76
  elif file_type == 'pdf':
77
  try:
 
80
  file_content = ""
81
  for page in pdf_reader.pages:
82
  file_content += page.extract_text() + "\n"
 
83
  except Exception as e:
84
  st.error(f"Error al leer el archivo PDF: {str(e)}")
85
  file_content = ""
 
89
  import docx
90
  doc = docx.Document(uploaded_file)
91
  file_content = "\n".join([para.text for para in doc.paragraphs])
 
92
  except Exception as e:
93
  st.error(f"Error al leer el archivo DOCX: {str(e)}")
94
  file_content = ""
95
 
96
+ # Remove these two lines to eliminate the preview
97
+ # st.write("**Vista previa del contenido:**")
98
+ # st.write(file_content[:500] + "..." if len(file_content) > 500 else file_content)
99
+
100
+ # Just show a success message instead
101
+ st.success(f"Archivo {file_type.upper()} cargado correctamente")
102
 
103
  # Set file type flag
104
  is_image = False
 
115
  "data": image_bytes
116
  }
117
  ]
 
118
 
119
  # Set file type flag
120
  is_image = True